package com.jeomo.common.core.util;

import com.jeomo.common.core.dto.PageResponseDto;
import com.jeomo.common.core.vo.PageResponseVo;
import org.springframework.beans.BeanUtils;

import java.util.ArrayList;
import java.util.List;
import java.util.function.Supplier;

/**
 * @Author: qbt
 * @Date: 2020/6/30 15:52
 * @Version 1.0
 */
public class BeanCopyUtil extends BeanUtils {

    /**
     * 如果source为空，则将返回null
     * T 应该实现一个空参构造函数
     * @param source
     * @param clazz
     * @param <T>
     * @return
     */
    public static <T> T copyProperties(Object source, Class<T> clazz) {
        if(source == null) {
            return null;
        }
        T target = null;
        try {
            target = clazz.newInstance();
            copyProperties(source, target);
        } catch (InstantiationException e) {
            e.printStackTrace();
        } catch (IllegalAccessException e) {
            e.printStackTrace();
        }
        return target;
    }

    /**
     * 分页查询转换
     * 主要是转换里面的items
     * @param dto
     * @param clazz
     * @return
     */
    public static  <T> PageResponseVo copyPageProperties(PageResponseDto dto, Class<T> clazz) {
        PageResponseVo<T> vo = new PageResponseVo<T>();
        vo.setTotal(dto.getTotal());
        List<?> items = dto.getItems();
        if(items == null) {
            return vo;
        }
        List voItems = new ArrayList();
        for (Object item : items) {
            voItems.add(copyProperties(item, clazz));
        }
        vo.setItems(voItems);
        return vo;
    }


    /**
     * 集合数据的拷贝
     * @param sources: 数据源类
     * @param target: 目标类::new(eg: UserVO::new)
     * @return
     */
    public static <S, T> List<T> copyListProperties(List<S> sources, Supplier<T> target) {
        return copyListProperties(sources, target, null);
    }

    /**
     * 带回调函数的集合数据的拷贝（可自定义字段拷贝规则）
     * @param sources: 数据源类
     * @param target: 目标类::new(eg: UserVO::new)
     * @param callBack: 回调函数
     * @return
     */
    public static <S, T> List<T> copyListProperties(List<S> sources, Supplier<T> target, BeanCopyUtilCallBack<S, T> callBack) {
        if(sources == null) {
            return null;
        }
        List<T> list = new ArrayList<>(sources.size());
        for (S source : sources) {
            T t = target.get();
            copyProperties(source, t);
            list.add(t);
            if (callBack != null) {
                // 回调
                callBack.callBack(source, t);
            }
        }
        return list;
    }

}
